# -*- coding: utf-8 -*-
import numpy as np


def AtomCenter_Pos(**kw):
    Atom_Center = []
    if 'Date' in kw and kw['Date'] == '20190726':
        # For 20190726
        Atom_Center.append(np.array([568, 544, 0, 0]))  # 0, Up
        Atom_Center.append(np.array([405, 504, 0, 0]))  # 1, Dn
    
    elif 'Date' in kw and kw['Date'] == '20191018':
        # For 20190727
        Atom_Center.append(np.array([588, 544, 0, 0]))  # 0, Up
        Atom_Center.append(np.array([405, 504, 0, 0]))  # 1, Dn
    # elif 'Date' in kw and kw['Date'] == '20200115':
    else:
        # For 20190727
        Atom_Center.append(np.array([665, 541, 0, 0]))  # 0, Up
        Atom_Center.append(np.array([476, 497, 0, 0]))  # 1, Dn
    Atom_Center.append(np.array([343, 230, 0, 0]))  # Background
    return Atom_Center


def CutWidths():
    # Cut_Width = np.zeros(3)
    # Cut_Width[0] = 200
    # Cut_Width[1] = 200
    # Cut_Width[2] = 100 # Background
    # Cut_Width = Cut_Width.astype(int)
    # Cut_Width = (200, 200, 100)
    Cut_Width = ([160, 250], [160, 250], 100)
    return Cut_Width
